package com.itheima.tlias.interceptor;

import com.itheima.tlias.utils.Constant;
import com.itheima.tlias.utils.JwtUtils;
import com.itheima.tlias.utils.UserThreadLocal;
import io.jsonwebtoken.Claims;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.HttpStatus;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Slf4j
public class LoginInterceptor implements HandlerInterceptor  {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)  {
        // 进入controller执行 先进入preHandler，如果这个方法返回true，进入controller，如果是false，不进入controller
        //配置器已实现 校验登录页面逻辑
        log.info("preHandle:{}", request.getRequestURI());
        //1. 获取请求url。
        String url = request.getRequestURI();
        //2. 判断请求url中是否包含login，如果包含，说明是登录操作，放行,配置中已实现。
        if (url.contains("/login")) {
            log.info("--------> 当前请求是登录请求, 直接放行");
            return true;
        }
        //3. 获取请求头中的令牌（token）。
        String jwt = request.getHeader("token");
        //4. 判断令牌是否存在，如果不存在，响应401。
        if (!StringUtils.hasLength(jwt)) {
            log.info("令牌不存在, 响应401");
            response.setStatus(HttpStatus.SC_UNAUTHORIZED);
            return false;
        }
        //5. 解析token，如果解析失败，响应401 。
        try {
            Claims claims = JwtUtils.parseJWT(jwt);
            Integer id = claims.get(Constant.ID, Integer.class);
            String name = claims.get(Constant.NAME, String.class);
            // 向当前线程中保存了一个UserInfo
            UserThreadLocal.setUserInfo(new UserThreadLocal.UserInfo(id, name ));
        } catch (Exception e) {
            log.error("令牌解析出错{}", e.getMessage());
            response.setStatus(401);
            return false;
        }
        //6. 放行。
        log.info("当前请求, 携带了令牌 : {}, 且令牌校验通过~", jwt);
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView)   {
        // controller执行完成以以后，会执行这个方法。
        // 如果controll执行的时候出现异常，这个方法不执行
        log.info("postHandle:{}", request.getRequestURI());
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)   {
        // 页面渲染完成以后再执行
        // 就算controll执行的时候出现异常，这个方法也要执行
        // 页面渲染完成以后再执行
        // 就算controller执行的时候出现异常，这个方法也要执行
        log.info("afterCompletion:{}", request.getRequestURI());
        // 从当前线程中删除UserInfo
        UserThreadLocal.removeUserInfo();
    }
}
